package com.example.permissionservice.config;

import com.myprojects.common.pojo.User;
import com.myprojects.common.pojo.UserWithRole;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.*;
import org.springframework.data.redis.serializer.RedisSerializer;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import javax.annotation.Resource;
import java.io.Closeable;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.TimeUnit;

/**
 * Redis工具类
 */
@Component
public class RedisManager {

    @Resource
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 指定缓存失效时间
     *
     * @param key  键
     * @param time 时间(秒)
     * @return 30
     */
    public boolean expire(String key, long time) {
        try {
            if (time > 0) {
                redisTemplate.expire(key, time, TimeUnit.SECONDS);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 根据key 获取过期时间
     *
     * @param key 键 不能为null
     * @return 时间(秒) 返回0代表为永久有效
     */
    public long getExpire(String key) {
        return redisTemplate.getExpire(key, TimeUnit.SECONDS);
    }

    /**
     * 判断key是否存在
     *
     * @param key 键
     * @return true 存在 false不存在
     */
    public boolean hasKey(String key) {
        try {
            return redisTemplate.hasKey(key);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 删除缓存
     *
     * @param key 可以传一个值 或多个
     */
    @SuppressWarnings("unchecked")
    public void deleteKey(String... key) {
        if (key != null && key.length > 0) {
            if (key.length == 1) {
                redisTemplate.delete(key[0]);
            } else {
                redisTemplate.delete(CollectionUtils.arrayToList(key));
            }
        }
    }

    /**
     * 普通缓存获取
     *
     * @param key 键
     * @return 值
     */
    public Object getObject(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 普通缓存放入
     * @param key   键
     * @param value 值
     * @return true成功 false失败
     */
    public boolean setObject(String key, Object value) {
        try {
            redisTemplate.opsForValue().set(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key   键
     * @param value 值
     * @param time  时间(秒) time要大于0 如果time小于等于0 将设置无限期
     * @return true成功 false 失败
     */
    public boolean setObject(String key, Object value, long time) {
        try {
            if (time > 0) {
                redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
            } else {
                setObject(key, value);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 用于查询整个redis数据库，返回的是一个带查询结果的HashMap<br/>
     *
     * @param pattern 数据库中键的模糊名字 不能为null
     * @return 各个redis数据库里的的键值对
     */
    public HashMap<String, Object> scanByPattern(String pattern) {
        //键的列表
        List<String> keys = new ArrayList<>();
        //值的列表
        List<Object> values = new ArrayList<>();
        //键-值对们
        HashMap<String, Object> hashMap = new HashMap();
        Cursor cursor = null;
        try {
            RedisSerializer<UserWithRole> redisSerializer = (RedisSerializer<UserWithRole>) redisTemplate.getKeySerializer();
            ScanOptions options = ScanOptions.scanOptions()
                    .count(Long.MAX_VALUE) //这里指定每次扫描key的数量
                    .match(pattern).build();
            cursor = (Cursor) redisTemplate.executeWithStickyConnection(
//                redisConnection ->
//                        new ConvertingCursor<>(redisConnection.scan(options), redisSerializer::deserialize)
                    new RedisCallback<Closeable>() {
                        @Override
                        public Closeable doInRedis(RedisConnection redisConnection) throws DataAccessException {
                            return new ConvertingCursor<>(redisConnection.scan(options), redisSerializer::deserialize);
                        }
                    }
            );
            while(cursor.hasNext()){
                keys.add(cursor.next().toString());
            }
            //获取
            values = redisTemplate.opsForValue().multiGet(keys);

            //装配结果集合
            for (int i = 0; i < keys.size(); i++) {
                hashMap.put(keys.get(i), values.get(i));
            }

        }catch (Exception e){
            e.printStackTrace();
        }finally {
            try {
                //切记这里一定要关闭，否则会耗尽连接数。
                cursor.close();
            }catch (IOException e){
                e.printStackTrace();
            }
        }

        return hashMap;
    }

    /**
     * 获取hashKey对应的所有键值
     * @param key 键
     * @return 对应的多个键值
     */
    public Map<Object, Object> getHashMap(String key) {
        return redisTemplate.opsForHash().entries(key);
    }

    /**
     * HashSet
     * @param key 键
     * @param map 对应多个键值
     * @return true 成功 false 失败
     */
    public boolean setHashMap(String key, Map<String, Object> map) {
        try {
            redisTemplate.opsForHash().putAll(key, map);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 向一张hash表中放入数据,如果该表不存在则创建
     * @param hashKey hash表名，不能为null
     * @param itemKey 表中的对象的键，不能为null
     * @param item    表中的对象的值，不能为null
     * @return true 成功 false失败
     */
    public boolean hashSet(String hashKey, String itemKey, Object item) {
        if (hashKey == null || itemKey == null || item == null)
            return false;
        try {
            redisTemplate.opsForHash().put(hashKey, itemKey, item);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 判断hash表中是否有该项的值
     * @param key  键 不能为null
     * @param item 项 不能为null
     * @return true 存在 false不存在
     */
    public boolean hashHasKey(String key, String item) {
        return redisTemplate.opsForHash().hasKey(key, item);
    }

    /**
     * HashGet
     * @param hashKey hash表名 不能为null
     * @param itemKey 表中键的名字 不能为null
     * @return 值
     */
    public Object hashGet(String hashKey, String itemKey) {
        return redisTemplate.opsForHash().get(hashKey, itemKey);
    }

    /**
     * 用于查询哈希表，返回的是一个查询结果的游标<br/>
     * 警告！返回的Cursor必须被关闭，否则会占用数据库连接！
     *
     * @param hashKey hash表名 不能为null
     * @param itemKey 表中键的模糊名字 不能为null
     * @return 值
     */
    public Cursor<Map.Entry<Object, Object>> hashGetScanCursor(String hashKey, String itemKey) {
        Cursor<Map.Entry<Object, Object>> cursor = null;
        try{
            cursor = redisTemplate.opsForHash().scan(
                    hashKey,
                    ScanOptions.scanOptions().match(itemKey).build());
        }catch (Exception e) {
            e.printStackTrace();
        }
        return cursor;
    }

    /**
     * 删除hash表中的值
     * @param key  hash表名 不能为null
     * @param item 表内需要删除的项 可以是多个 不能为null
     */
    public void hashDelete(String key, Object... item) {
        redisTemplate.opsForHash().delete(key, item);
    }



}

